#!/bin/bash

# 设置安装包名称
ANACONDA_INSTALLER="Anaconda3-2023.09-0-Linux-x86_64.sh"

# 检查当前目录是否已经存在 Anaconda 安装包
if [ -f "$ANACONDA_INSTALLER" ]; then
    echo "检测到当前目录已存在 Anaconda 安装包，跳过下载。"
else
    echo "未找到 Anaconda 安装包，正在下载..."
    wget https://repo.anaconda.com/archive/"$ANACONDA_INSTALLER"
    if [ $? -ne 0 ]; then
        echo "错误：下载 Anaconda 安装包失败。"
        exit 1
    fi
fi

# 设置安装路径为当前目录下的 anaconda3
INSTALL_DIR="$(pwd)/anaconda3"

# 检查是否已经安装过 Anaconda
if [ -d "$INSTALL_DIR" ]; then
    echo "检测到 Anaconda 已安装到 $INSTALL_DIR，跳过安装。"
else
    # 安装 Anaconda
    echo "正在安装 Anaconda 到 $INSTALL_DIR..."
    bash "$ANACONDA_INSTALLER" -b -p "$INSTALL_DIR"
    if [ $? -ne 0 ]; then
        echo "错误：Anaconda 安装失败。"
        exit 1
    fi
fi

# 更改 Anaconda 安装目录权限
echo "更改 Anaconda 安装目录权限..."
sudo chown -R $USER:$USER "$INSTALL_DIR" || { echo "错误：更改目录权限失败。"; exit 1; }
sudo chmod -R u+rw "$INSTALL_DIR" || { echo "错误：更改目录权限失败。"; exit 1; }

# 使用 conda init 初始化环境（针对当前用户）
echo "配置 Anaconda 环境..."
"$INSTALL_DIR/bin/conda" init bash
if [ $? -ne 0 ]; then
    echo "错误：conda init 初始化失败。"
    exit 1
fi

# 加载新的环境变量
source ~/.bashrc

# 检查 Python 和 Conda 版本
echo "检查 Python 和 Conda 版本..."
"$INSTALL_DIR/bin/python" --version
"$INSTALL_DIR/bin/conda" --version

echo "Anaconda 安装完成。请运行以下命令确认 Python 版本："
echo "  $INSTALL_DIR/bin/python --version"
echo "如果 Python 版本未变更，请关闭终端并重新打开，或者手动运行 'source ~/.bashrc'。"